package com.lzy.service.impl;

import com.lzy.annotation.DataScope;
import com.lzy.constant.Constants;
import com.lzy.context.LocalUser;
import com.lzy.service.ISysDeptService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;

import com.lzy.dao.SysDeptDao;
import com.lzy.entity.SysDeptEntity;

import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

@Service("sysDeptService")
public class SysDeptService extends ServiceImpl<SysDeptDao, SysDeptEntity> implements ISysDeptService {

    @Autowired
    private SysDeptDao mapper;

    @Override
    public List<SysDeptEntity> selectPageList(SysDeptEntity sysDeptEntity) {

        return mapper.selectPageList(sysDeptEntity);
    }

    @Override
    @DataScope(deptAlias = "d")
    public List<SysDeptEntity> selectPageTreeList(SysDeptEntity sysDeptEntity) {
        List<SysDeptEntity> sysDeptEntities = mapper.selectPageList(sysDeptEntity);

        return sysDeptEntities.stream().filter(item -> {
            for (SysDeptEntity deptEntity : sysDeptEntities) {
                if (item.getParentId().equals(deptEntity.getDeptId())) {
                    deptEntity.addChildren(item);
                    return false;
                }
            }

            return true;
        }).collect(Collectors.toList());
    }
}